Numpy array indexing with complete vector